#Run this file use command: python setup.py build_ext --inplace
from distutils.core import setup
from Cython.Build import cythonize
import numpy as np

setup(
    name="AES accelerator",
    ext_modules=cythonize(["crypto_kit/cython/AES_accelerator.pyx"]),
    include_dirs=[np.get_include()]
)

setup(
    name="SM4 accelerator",
    ext_modules=cythonize(["crypto_kit/cython/SM4_accelerator.pyx"]),
    include_dirs=[np.get_include()]
)